package com.example.swaggerdemo.controller;

import com.example.swaggerdemo.core.R;
import com.example.swaggerdemo.entity.UserEntity;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * swagger 用户测试方法
 *
 * @author chenzz
 */
@Api(tags = "用户信息")
@RestController
@RequestMapping("/test/user")
public class TestController {

    private final static Map<Integer, UserEntity> users = new LinkedHashMap<>();

    {
        users.put(1, new UserEntity(1, "admin", "55555", "15888888888"));
        users.put(2, new UserEntity(2, "chenzz", "66666", "15666666666"));
    }

    @ApiOperation("获取用户列表")
    @GetMapping
    public R userList() {
        List<UserEntity> userList = new ArrayList<>(users.values());
        return R.ok().data(userList);
    }

    @ApiOperation("获取用户详细")
    @GetMapping("/{userId}")
    public R getUser(@PathVariable Integer userId) {
        if (!users.isEmpty() && users.containsKey(userId)) {
            return R.ok().data(users.get(userId));
        }
        return R.error("用户不存在");
    }

    @ApiOperation("新增用户")
    @PostMapping
    public R addSave(@RequestBody UserEntity user) {
        if (StringUtils.isEmpty(user) || StringUtils.isEmpty(user.getUserId())) {
            return R.error("用户ID不能为空");
        }
        if (users.containsKey(user.getUserId())) {
            return R.error("用户已存在");
        }
        return R.ok().data(users.put(user.getUserId(), user));
    }

    @ApiOperation("更新用户")
    @PutMapping
    public R update(@RequestBody UserEntity user) {
        if (StringUtils.isEmpty(user) || StringUtils.isEmpty(user.getUserId())) {
            return R.error("用户ID不能为空");
        }
        if (users.isEmpty() || !users.containsKey(user.getUserId())) {
            return R.error("用户不存在");
        }
        users.remove(user.getUserId());
        return R.ok().data(users.put(user.getUserId(), user));
    }

    @ApiOperation("删除用户信息")
    @DeleteMapping("/{userId}")
    public R delete(@PathVariable Integer userId) {
        if (!users.isEmpty() && users.containsKey(userId)) {
            users.remove(userId);
            return R.ok();
        }
        return R.error("用户不存在");
    }
}

